/*
 * Copyright (c) 2008-2011, Petr Panteleyev
 * All rights reserved.
 *
 * Redistribution and use in source and binary forms, with or without modification,
 * are permitted provided that the following conditions are met:
 *
 *    1. Redistributions of source code must retain the above copyright notice,
 *       this list of conditions and the following disclaimer.
 *    2. Redistributions in binary form must reproduce the above copyright notice,
 *       this list of conditions and the following disclaimer in the documentation
 *       and/or other materials provided with the distribution.
 *    3. The name of the author may not be used to endorse or promote products
 *       derived from this software without specific prior written permission.
 *
 * THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS'' AND ANY EXPRESS OR IMPLIED
 * WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY
 * AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE AUTHOR
 * BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
 * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES;
 * LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
 * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING
 * NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN
 * IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
 */

package org.panteleyev.html;

import javax.swing.event.DocumentEvent;
import javax.swing.text.AttributeSet;
import javax.swing.text.Document;
import javax.swing.text.Element;
import javax.swing.text.MutableAttributeSet;
import javax.swing.text.StyleConstants;
import javax.swing.text.html.HTML;
import javax.swing.text.html.HTMLDocument;
import javax.swing.text.html.StyleSheet;

public class MutableHTMLDocument extends HTMLDocument {
    public MutableHTMLDocument(StyleSheet styles) {
        super(styles);
    }

    public Element getElementByTag(HTML.Tag tag) {
        Element root = getDefaultRootElement();
        return getElementByTag(root, tag);
    }

    public Element getElementByTag(Element parent, HTML.Tag tag) {
        if ((parent == null) || (tag == null)) {
            return null;
        }

        for (int k = 0; k < parent.getElementCount(); k++) {
            Element child = parent.getElement(k);
            if (child.getAttributes().getAttribute(StyleConstants.NameAttribute).equals(tag)) {
                return child;
            }

            Element e = getElementByTag(child, tag);
            if (e != null) {
                return e;
            }
        }

        return null;
    }

    public String getTitle() {
        return (String)getProperty(Document.TitleProperty);
    }

    public void addAttributes(Element e, AttributeSet attributes) {
        if ((e == null) || (attributes == null)) {
            return;
        }

        try {
            writeLock();
            MutableAttributeSet mAttr = (MutableAttributeSet)e.getAttributes();
            mAttr.addAttributes(attributes);
            fireChangedUpdate(new DefaultDocumentEvent(0, getLength(), DocumentEvent.EventType.CHANGE));
        } finally {
            writeUnlock();
        }
    }
}
